package com.dayangxiaoen.pms.utils;

import io.jsonwebtoken.Jwt;
import io.jsonwebtoken.JwtBuilder;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.junit.Test;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

/**
 * @author dayangxiaoen
 *
 * 生成JWT Token 的工具类
 *
 * 使用秘钥为 666#dayangxiaoen#666 的秘钥
 */
public class JWTUtils {


    // 设置生成 token 的 秘钥
    private static final String jwtToken = "666#dayangxiaoen#666";

    /**
     *
     * @param userId
     * @return String
     *
     *  根据 用户ID 创建token
     */
    public static String createToken(Long userId){
        Map<String,Object> claims = new HashMap<>();
        claims.put("userId",userId);
        JwtBuilder jwtBuilder = Jwts.builder()
                .signWith(SignatureAlgorithm.HS256, jwtToken) // 签发算法，秘钥为jwtToken
                .setClaims(claims) // body数据，要唯一，自行设置
                .setIssuedAt(new Date()) // 设置签发时间
                .setExpiration(new Date(System.currentTimeMillis() + 24 * 60 * 60 * 60 * 1000));// 一天的有效时间
        String token = jwtBuilder.compact();
        return token;
    }

    /**
     *
     * @param token
     * @return Map<String, Object>
     *
     * 校验一个 token 是否合法
     * 根据 token 可以获得用户id
     */
    public static Map<String, Object> checkToken(String token){
        try {
            Jwt parse = Jwts.parser().setSigningKey(jwtToken).parse(token);
            return (Map<String, Object>) parse.getBody();
        }catch (Exception e){
            e.printStackTrace();
        }
        return null;

    }

    /**
     * 单元测试
     * 测试相关工具类是否能正确运行
     */
    @Test
    public  void TestToken() {
        String token = JWTUtils.createToken(100L);
        System.out.println(token);
        Map<String, Object> stringObjectMap = JWTUtils.checkToken(token);
        System.out.println(stringObjectMap.get("userId"));

    }

}
